source env_npu.sh

export RANK_SIZE=8
rm -f nohup.out

for((RANK_ID=0;RANK_ID<RANK_SIZE;RANK_ID++));
do
    nohup python train_8p.py \
        --batch_size 8  \
        --model 'resnet152'  \
        --multigpu True  \
        --device_id ${RANK_ID}  \
        --npus ${RANK_SIZE} \
        --lr 5e-4 >> train.txt 2>&1 &
done

